package com.indusfo.spc.controller;

import com.indusfo.spc.aspect.SystemControllerLog;
import com.indusfo.spc.pojo.ExcelErrorData;
import com.indusfo.spc.pojo.Site;
import com.indusfo.spc.pojo.SiteMat;
import com.indusfo.spc.service.SiteMatService;
import com.indusfo.spc.untlis.ExcelUtil;
import com.indusfo.spc.untlis.GsonUtil;
import com.indusfo.spc.vo.JSONObject;
import org.apache.commons.lang3.StringUtils;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.List;

/**
 * 库位物料控制层
 */
@Controller
@RequestMapping("/service/siteMat")
public class SiteMatController {

    @Autowired
    private SiteMatService siteMatService;

    /**
     * 查询
     * @return com.indusfo.spc.vo.JSONObject
     */
    @RequestMapping(value="/querySiteMat")
    @SystemControllerLog(description="查询")
    @ResponseBody
    public JSONObject querySiteMat(SiteMat siteMat) {
        JSONObject json = siteMatService.selectSiteMat(siteMat);
        return json;
    }

    /**
     * 新增
     * @return com.indusfo.spc.vo.JSONObject
     */
    @RequestMapping(value="/insertSiteMat")
    @SystemControllerLog(description="新增")
    @ResponseBody
    public JSONObject insertSiteMat(SiteMat siteMat) {
        JSONObject json = siteMatService.insertSiteMat(siteMat);
        return json;
    }

    /**
     * 修改
     * @return
     */
    @RequestMapping(value = "/updateSiteMat")
    @SystemControllerLog(description="修改")
    @ResponseBody
    public JSONObject updateSiteMat(SiteMat siteMat){
        JSONObject json=siteMatService.updateSiteMat(siteMat);
        return json;
    }

    /**
     * 启用/停用/删除
     * @return com.indusfo.spc.vo.JSONObject
     */
    @RequestMapping(value="/definedSiteMat")
    @SystemControllerLog(description="启用/停用/删除")
    @ResponseBody
    public JSONObject definedSiteMat(Long[] siteMatIds, Integer dataState) {
        dataState = 2;
        JSONObject json = siteMatService.definedSiteMat(siteMatIds, dataState);
        return json;
    }

    @RequestMapping("/uploadFileSiteMat")
    @ResponseBody
    public JSONObject uploadFileSiteMat(@RequestParam("file") MultipartFile file, Boolean way, HttpServletRequest request, HttpServletResponse response) throws IOException, ParseException {

        //定义一个message对象用于向前台传输数据
        JSONObject jsonObject=new JSONObject();
        jsonObject.setStatus(JSONObject.SUCCESS);

        //设置错误集合的
        List<ExcelErrorData> errorDatas = new ArrayList<>();
        response.setContentType("text/html;charset=UTF-8");
        //判断上传文件格式是否正确
        Workbook wb = ExcelUtil.getWorkbook(file.getInputStream(), ExcelUtil.isExcel2007(file.getOriginalFilename()));
        if (wb == null) {
            errorDatas.add(ExcelErrorData.getExcelErrorData(0, 0, "文件解析错误"));
        } else {
            //获取excel文件的第一个sheet
            Sheet sheet = wb.getSheetAt(0);
            if (sheet == null) {
                jsonObject.setStatus(JSONObject.ERROR);
                errorDatas.add(ExcelErrorData.getExcelErrorData(0, 0, "未找到需要导入的表格"));
            } else {
                //获取最后一行的行号
                int rowCount = sheet.getLastRowNum();
                if (rowCount < 1) {
                    jsonObject.setStatus(JSONObject.ERROR);
                    errorDatas.add(ExcelErrorData.getExcelErrorData(0, "未找到需要导入的数据，请确认表格是否保存！！！"));
                } else {
                    List<SiteMat> siteMatList = new ArrayList<>();
                    for (int i = 1; i <= rowCount; i++) {
                        boolean needAdd = true;
                        //获取到每一行的每一列的数据
                        //物料名称
                        String matName =com.indusfo.spc.untlis.ExcelUtil.getCellValue(sheet, i, 0, false);
                        //库位名称
                        String siteName = com.indusfo.spc.untlis.ExcelUtil.getCellValue(sheet, i, 1, false);
                        //最小库存
                        String siteMatNum = com.indusfo.spc.untlis.ExcelUtil.getCellValue(sheet, i, 2, false);
                        SiteMat siteMat = new SiteMat(matName,siteMatNum,siteName);
                        int m = i + 1;
                        //将excel中的每一行数据存储进行检查
                        String result = siteMatService.checkNoRepeatExcel(siteMat);
                        if(result!=null){
                            errorDatas.add(ExcelErrorData.getExcelErrorData(i, result));
                            needAdd=false;
                        }

                        if (needAdd) {
                            siteMatList.add(siteMat);
                        } else {
                            jsonObject.setStatus(JSONObject.ERROR);
                            jsonObject.setMsg("导入数据中存在异常信息");
                        }
                    }
                    //根据这些来判断是否是中止导入还是跳过继续进行导入
                    if (jsonObject.getStatus().equals(JSONObject.ERROR) && !way) {
                        jsonObject.setMsg(jsonObject.getMsg()+ ",中止导入");
                    } else {
                        if (siteMatList.size() > 0) {
                            try {
                                siteMatService.updateListSiteMat(siteMatList);
                                //当程序执行到这里时,判断信息是否为空为空的话设置导入成功
                                jsonObject.setMsg(StringUtils.isBlank(jsonObject.getMsg()) ? "导入成功" : jsonObject.getMsg());
                                if ("导入成功".equals(jsonObject.getMsg())) {
                                    jsonObject.setStatus(JSONObject.SUCCESS);
                                }
                            } catch (Exception e) {
                                e.printStackTrace();
                                jsonObject.setMsg("导入失败");
                            }
                        }
                    }
                    //msg.setData(GsonUtil.toJson(errorDatas));
                }
            }
        }
        jsonObject.setData(GsonUtil.toJson(errorDatas));
        return JSONObject.build(JSONObject.SUCCESS,jsonObject.getMsg(),errorDatas);
    }
}
